tensor([[0., 1., 0., 1., 0.],
        [1., 0., 1., 0., 1.]], grad_fn=<RoundBackward1>)